package util;

import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.GraphicsConfiguration;
import java.awt.GraphicsDevice;
import java.awt.GraphicsEnvironment;
import java.awt.HeadlessException;
import java.awt.Image;
import java.awt.RenderingHints;
import java.awt.Transparency;
import java.awt.image.BufferedImage;
import java.awt.image.ColorModel;
import java.awt.image.PixelGrabber;
import java.awt.image.WritableRaster;

import javax.swing.ImageIcon;

import model.Sprite;

//a lot of this code is from:
//http://www.javalobby.org/articles/ultimate-image/#9
//http://www.exampledepot.com/egs/java.awt.image/image2buf.html

public class ImageUtil
{

	public static BufferedImage getScaledImage(BufferedImage bufferedImageSrc, double factor)
	{

		int w = bufferedImageSrc.getWidth();
		int h = bufferedImageSrc.getHeight();
		int width = (int) (factor * w);
		int height = (int) (factor * h);
		BufferedImage bufferedImageScaled = new BufferedImage(width, height, bufferedImageSrc.getType());
		Graphics2D g = bufferedImageScaled.createGraphics();
		g.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
		g.drawImage(bufferedImageSrc, 0, 0, width, height, 0, 0, w, h, null);
		g.dispose();
		return bufferedImageScaled;
	}

	public static BufferedImage getImageFlippedHorizontal(BufferedImage bufferedImageSrc)
	{
		int w = bufferedImageSrc.getWidth();
		int h = bufferedImageSrc.getHeight();
		BufferedImage flippedImage = new BufferedImage(w, h, bufferedImageSrc.getType());
		Graphics2D g = flippedImage.createGraphics();
		g.drawImage(bufferedImageSrc, 0, 0, w, h, w, 0, 0, h, null);
		g.dispose();
		return flippedImage;
	}

	/**
	 * Deep copy of BufferedImage
	 * 
	 * @param bi
	 * @return
	 */
	public static BufferedImage cloneBufferedImage(BufferedImage bufferedImageSrc)
	{
		ColorModel cm = bufferedImageSrc.getColorModel();
		boolean isAlphaPremultiplied = cm.isAlphaPremultiplied();
		WritableRaster raster = bufferedImageSrc.copyData(null);
		return new BufferedImage(cm, raster, isAlphaPremultiplied, null);
	}

	public static BufferedImage rotate(BufferedImage bufferedImagSrc, double angle)
	{
		int w = bufferedImagSrc.getWidth();
		int h = bufferedImagSrc.getHeight();
		BufferedImage rotatedImage = new BufferedImage(w, h, bufferedImagSrc.getType());
		Graphics2D g = rotatedImage.createGraphics();
		g.rotate(Math.toRadians(angle), w / 2, h / 2);
		g.drawImage(bufferedImagSrc, null, 0, 0);
		return rotatedImage;
	}

	public static BufferedImage toBufferedImage(Image image)
	{

		if (image instanceof BufferedImage)
		{
			return (BufferedImage) image;
		}

		// This code ensures that all the pixels in the image are loaded
		image = new ImageIcon(image).getImage();

		// Determine if the image has transparent pixels; for this method's
		// implementation, see Determining If an Image Has Transparent Pixels
		boolean hasAlpha = hasAlpha(image);

		// Create a buffered image with a format that's compatible with the screen
		BufferedImage bimage = null;
		GraphicsEnvironment ge = GraphicsEnvironment.getLocalGraphicsEnvironment();
		try
		{
			// Determine the type of transparency of the new buffered image
			int transparency = Transparency.OPAQUE;
			if (hasAlpha)
			{
				transparency = Transparency.BITMASK;
			}

			// Create the buffered image
			GraphicsDevice gs = ge.getDefaultScreenDevice();
			GraphicsConfiguration gc = gs.getDefaultConfiguration();
			bimage = gc.createCompatibleImage(image.getWidth(null), image.getHeight(null), transparency);
		}
		catch (HeadlessException e)
		{
			// The system does not have a screen
		}

		if (bimage == null)
		{
			// Create a buffered image using the default color model
			int type = BufferedImage.TYPE_INT_RGB;
			if (hasAlpha)
			{
				type = BufferedImage.TYPE_INT_ARGB;
			}
			bimage = new BufferedImage(image.getWidth(null), image.getHeight(null), type);
		}

		// Copy image to buffered image
		Graphics g = bimage.createGraphics();

		// Paint the image onto the buffered image
		g.drawImage(image, 0, 0, null);
		g.dispose();

		return bimage;
	}

	// This method returns true if the specified image has transparent pixels
	public static boolean hasAlpha(Image image)
	{
		// If buffered image, the color model is readily available
		if (image instanceof BufferedImage)
		{
			BufferedImage bimage = (BufferedImage) image;
			return bimage.getColorModel().hasAlpha();
		}

		// Use a pixel grabber to retrieve the image's color model;
		// grabbing a single pixel is usually sufficient
		PixelGrabber pg = new PixelGrabber(image, 0, 0, 1, 1, false);
		try
		{
			pg.grabPixels();
		}
		catch (InterruptedException e)
		{
		}

		// Get the image's color model
		ColorModel cm = pg.getColorModel();
		return cm.hasAlpha();
	}
}
